import string
import pymysql
import urllib.request
from pyexpat import features
from urllib.parse import quote

from bs4 import BeautifulSoup

city=input("请输入城市:")
code={
    '北京': '530',
    '上海': '538',
    '广州': '763',
    '深圳': '765',
    '天津': '531',
    '武汉': '736',
    '西安': '854',
    '成都': '801',
    '沈阳': '599',
    '南京': '635',
    '杭州': '653',
    '苏州': '639',
    '重庆': '551',
    '长沙': '749',
    '厦门': '682',
    '南昌': '691'
}
city=code[city]
type=input("请输入岗位:")

url="https://sou.zhaopin.com/?jl="+city+"&kw="+type+"&p=1"

url=quote(url,safe=string.printable)
res=urllib.request.urlopen(url)
print(url)

content=res.read().decode()
#print(content)

soup= BeautifulSoup(content,features="lxml")

jobitem=soup.select(".joblist-box__item")
print(jobitem)

job_list=[]
for job in jobitem:
    job_dic={}
    job_name=job.select(".jobinfo__name")[0].get_text()
    job_sal=job.select(".jobinfo__salary")[0].get_text()
    job_sal=job_sal.strip().replace(" ","").replace("\n","")
    print(job_sal)
    if '天' in job_sal:
        job_sal=int(job_sal.split("-")[0])*30
    elif('千' in job_sal):
        job_sal = int(job_sal.split("千")[0]) * 1000
    elif ('万' in job_sal):
        job_sal = float(job_sal.split("万")[0]) * 10000
    else:
        job_sal=5000
    company_name=job.select(".companyinfo__name")[0].get_text()
    job_msg=job.select(".jobinfo__other-info-item")
    job_place=job_msg[0].get_text()[1:3]
    job_exp=job_msg[1].get_text()
    job_education=job_msg[2].get_text()

    job_dic["job_name"]=job_name
    job_dic["job_sal"] = job_sal
    job_dic["company_name"] = company_name
    job_dic["job_place"] = job_place
    job_dic["job_exp"] = job_exp
    job_dic["job_education"] = job_education
    print(job_dic)
    job_list.append(job_dic)

db = pymysql.connect(
    host="localhost",
    port=3306,
    user="root",
    database="crawlerdemo",
    password="",
    charset="utf8"
)
cursor=db.cursor()
for job in job_list:
    sql="insert into job values (null,'%s','%s','%s','%s','%s','%s')"%(job["job_name"],job["job_sal"],job["company_name"],job["job_place"],job["job_exp"],job["job_education"])
    try:
        cursor.execute(sql)
    except Exception as e:
        print(e)
        db.rollback()
db.commit()
cursor.close()
db.close()